# 编写一个用户登陆程序

user_data = dict()


def show_menu():
    prompt = """
|--- 新建用户：N/n ---|
|--- 登录账号：E/e ---|
|--- 推出程序：Q/q ---|
|--- 请输入指令代码：
    """
    choice = input(prompt)
    while True:
        if choice not in "NnEeQq":
            print("您输入的指令有误，请重新输入：")
        if choice == 'q' or choice == 'Q':
            print("感谢您的使用！")
            exit(1)
        if choice == 'n' or choice == 'N':
            # 新建用户
            new_user()
        if choice == 'E' or choice == 'e':
            # 老用户登陆
            old_user()


def new_user():
    user_name = input("请输入要新建用户名：")
    loop = True
    while loop:
        if user_name not in user_data:
            user_data[user_name] = input("请输入密码：")
            print("新建用户信息成功！您可以使用此用户名进行登陆！")
            break
        else:
            user_name = input("此用户名已存在，请重新输入：")
            continue
    show_menu()


def old_user():
    user_name = input("请输入用户名：")
    loop = True
    while loop:
        if user_name in user_data:
            if user_data[user_name] == input("请输入密码："):
                print("尊敬的用户%s" % user_name + ",欢迎您！")
                break
            else:
                print("密码输入错误，请重试!")
                continue
        else:
            user_name = input("该用户名不存在，请重新输入:")
            continue
    show_menu()


if __name__ == "__main__":
    show_menu()